import pyciv.hexagon

class UnitType(object):
    def __init__(self, name, actionpoints):
        self.Name = name
        self.ActionPoints = actionpoints

Swordsmen = UnitType("Swordsmen", 5)

class Unit(object):
    def __init__(self, unittype, owner):
        self.Type = unittype
        self.Owner = owner
        self.Strength = 100
        self.Direction = pyciv.hexagon.RIGHT
        self.ActionPoints = self.Type.ActionPoints

    def __str__(self):
        return str(self.Strength)

    def NextTurn(self):
        self.ActionPoints = self.Type.ActionPoints
